/**
 * Copyright (C) 2011 Whisper Systems
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.appspot.eusms.notifications;

import android.app.Notification;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.content.Context;
import android.content.Intent;
import android.database.Cursor;
import android.graphics.BitmapFactory;
import android.graphics.Color;
import android.media.AudioManager;
import android.media.MediaPlayer;
import android.net.Uri;
import android.support.v4.app.NotificationCompat;
import android.support.v4.app.NotificationCompat.BigTextStyle;
import android.support.v4.app.NotificationCompat.InboxStyle;
import android.text.Spannable;
import android.text.SpannableString;
import android.text.SpannableStringBuilder;
import android.text.TextUtils;
import android.text.style.StyleSpan;
import android.util.Log;

import com.appspot.eusms.RoutingActivity;
import com.appspot.eusms.database.DatabaseFactory;
import com.appspot.eusms.database.MmsSmsDatabase;
import com.appspot.eusms.database.PushDatabase;
import com.appspot.eusms.database.model.MessageRecord;
import com.appspot.eusms.recipients.Recipient;
import com.appspot.eusms.recipients.RecipientFactory;
import com.appspot.eusms.recipients.RecipientFormattingException;
import com.appspot.eusms.recipients.Recipients;
import com.appspot.eusms.util.TextSecurePreferences;

import com.appspot.eusms.R;

import org.whispersystems.textsecure.crypto.MasterSecret;
import org.whispersystems.textsecure.push.IncomingPushMessage;

import java.io.IOException;
import java.util.List;

/**
 * Handles posting system notifications for new messages.
 *
 * @author Moxie Marlinspike
 */

public class MessageNotifier {

    public static final int NOTIFICATION_ID = 1338;

    private volatile static long visibleThread = -1;

    public static void setVisibleThread(long threadId) {
        visibleThread = threadId;
    }

    public static void notifyMessageDeliveryFailed(Context context, Recipients recipients, long threadId) {
        if (visibleThread == threadId) {
            sendInThreadNotification(context);
        } else {
            Intent intent = new Intent(context, RoutingActivity.class);
            intent.setFlags(Intent.FLAG_ACTIVITY_SINGLE_TOP);
            intent.putExtra("recipients", recipients);
            intent.putExtra("thread_id", threadId);
            intent.setData((Uri.parse("custom://" + System.currentTimeMillis())));

            NotificationCompat.Builder builder = new NotificationCompat.Builder(context);
            builder.setSmallIcon(R.drawable.icon_notification);
            builder.setLargeIcon(BitmapFactory.decodeResource(context.getResources(),
                    R.drawable.ic_list_alert_sms_failed));
            builder.setContentTitle(context.getString(R.string.MessageNotifier_message_delivery_failed));
            builder.setContentText(context.getString(R.string.MessageNotifier_failed_to_deliver_message));
            builder.setTicker(context.getString(R.string.MessageNotifier_error_delivering_message));
            builder.setContentIntent(PendingIntent.getActivity(context, 0, intent, 0));
            builder.setAutoCancel(true);
            setNotificationAlarms(context, builder, true);

            ((NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE))
                    .notify((int) threadId, builder.build());
        }
    }


    public static void updateNotification(Context context, MasterSecret masterSecret) {
        if (!TextSecurePreferences.isNotificationsEnabled(context)) {
            return;
        }

        updateNotification(context, masterSecret, false);
    }

    public static void updateNotification(Context context, MasterSecret masterSecret, long threadId) {
        if (!TextSecurePreferences.isNotificationsEnabled(context)) {
            return;
        }

        if (visibleThread == threadId) {
            DatabaseFactory.getThreadDatabase(context).setRead(threadId);
            sendInThreadNotification(context);
        } else {
            updateNotification(context, masterSecret, true);
        }
    }

    private static void updateNotification(Context context, MasterSecret masterSecret, boolean signal) {
        Cursor telcoCursor = null;
        Cursor pushCursor = null;

        try {
            telcoCursor = DatabaseFactory.getMmsSmsDatabase(context).getUnread();
            pushCursor = DatabaseFactory.getPushDatabase(context).getPending();

            if ((telcoCursor == null || telcoCursor.isAfterLast()) &&
                    (pushCursor == null || pushCursor.isAfterLast())) {
                ((NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE))
                        .cancel(NOTIFICATION_ID);
                return;
            }

            NotificationState notificationState = constructNotificationState(context, masterSecret, telcoCursor);

            appendPushNotificationState(context, masterSecret, notificationState, pushCursor);

            if (notificationState.hasMultipleThreads()) {
                sendMultipleThreadNotification(context, masterSecret, notificationState, signal);
            } else {
                sendSingleThreadNotification(context, masterSecret, notificationState, signal);
            }
        } finally {
            if (telcoCursor != null) telcoCursor.close();
            if (pushCursor != null) pushCursor.close();
        }
    }

    private static void sendSingleThreadNotification(Context context,
                                                     MasterSecret masterSecret,
                                                     NotificationState notificationState,
                                                     boolean signal) {
        if (notificationState.getNotifications().isEmpty()) {
            ((NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE))
                    .cancel(NOTIFICATION_ID);
            return;
        }

        List<NotificationItem> notifications = notificationState.getNotifications();
        NotificationCompat.Builder builder = new NotificationCompat.Builder(context);
        Recipient recipient = notifications.get(0).getIndividualRecipient();

        builder.setSmallIcon(R.drawable.icon_notification);
        builder.setLargeIcon(recipient.getContactPhoto());
        builder.setContentTitle(recipient.toShortString());
        builder.setContentText(notifications.get(0).getText());
        builder.setContentIntent(notifications.get(0).getPendingIntent(context));
        builder.setContentInfo(String.valueOf(notificationState.getMessageCount()));

        if (masterSecret != null) {
            builder.addAction(R.drawable.check, context.getString(R.string.MessageNotifier_mark_as_read),
                    notificationState.getMarkAsReadIntent(context, masterSecret));
        }

        SpannableStringBuilder content = new SpannableStringBuilder();

        for (NotificationItem item : notifications) {
            content.append(item.getBigStyleSummary());
            content.append('\n');
        }

        builder.setStyle(new BigTextStyle().bigText(content));

        setNotificationAlarms(context, builder, signal);

        if (signal) {
            builder.setTicker(notifications.get(0).getTickerText());
        }

        ((NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE))
                .notify(NOTIFICATION_ID, builder.build());
    }

    private static void sendMultipleThreadNotification(Context context,
                                                       MasterSecret masterSecret,
                                                       NotificationState notificationState,
                                                       boolean signal) {
        List<NotificationItem> notifications = notificationState.getNotifications();
        NotificationCompat.Builder builder = new NotificationCompat.Builder(context);

        builder.setSmallIcon(R.drawable.icon_notification);
        builder.setLargeIcon(BitmapFactory.decodeResource(context.getResources(),
                R.drawable.icon_notification));
        builder.setContentTitle(String.format(context.getString(R.string.MessageNotifier_d_new_messages),
                notificationState.getMessageCount()));
        builder.setContentText(String.format(context.getString(R.string.MessageNotifier_most_recent_from_s),
                notifications.get(0).getIndividualRecipientName()));
        builder.setContentIntent(PendingIntent.getActivity(context, 0, new Intent(context, RoutingActivity.class), 0));

        builder.setContentInfo(String.valueOf(notificationState.getMessageCount()));

        if (masterSecret != null) {
            builder.addAction(R.drawable.check, context.getString(R.string.MessageNotifier_mark_all_as_read),
                    notificationState.getMarkAsReadIntent(context, masterSecret));
        }

        InboxStyle style = new InboxStyle();

        for (NotificationItem item : notifications) {
            style.addLine(item.getTickerText());
        }

        builder.setStyle(style);

        setNotificationAlarms(context, builder, signal);

        if (signal) {
            builder.setTicker(notifications.get(0).getTickerText());
        }

        ((NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE))
                .notify(NOTIFICATION_ID, builder.build());
    }

    private static void sendInThreadNotification(Context context) {
        try {
            if (!TextSecurePreferences.isInThreadNotifications(context)) {
                return;
            }

            String ringtone = TextSecurePreferences.getNotificationRingtone(context);

            if (ringtone == null)
                return;

            Uri uri = Uri.parse(ringtone);
            MediaPlayer player = new MediaPlayer();
            player.setAudioStreamType(AudioManager.STREAM_NOTIFICATION);
            player.setDataSource(context, uri);
            player.setLooping(false);
            player.setVolume(0.25f, 0.25f);
            player.prepare();

            final AudioManager audioManager = ((AudioManager) context.getSystemService(Context.AUDIO_SERVICE));

            audioManager.requestAudioFocus(null, AudioManager.STREAM_NOTIFICATION,
                    AudioManager.AUDIOFOCUS_GAIN_TRANSIENT_MAY_DUCK);

            player.setOnCompletionListener(new MediaPlayer.OnCompletionListener() {
                @Override
                public void onCompletion(MediaPlayer mp) {
                    audioManager.abandonAudioFocus(null);
                }
            });

            player.start();
        } catch (IOException ioe) {
            Log.w("MessageNotifier", ioe);
        }
    }

    private static void appendPushNotificationState(Context context,
                                                    MasterSecret masterSecret,
                                                    NotificationState notificationState,
                                                    Cursor cursor) {
        if (masterSecret != null) return;

        PushDatabase.Reader reader = null;
        IncomingPushMessage message;

        try {
            reader = DatabaseFactory.getPushDatabase(context).readerFor(cursor);

            while ((message = reader.getNext()) != null) {
                Recipient recipient;

                try {
                    recipient = RecipientFactory.getRecipientsFromString(context, message.getSource(), false).getPrimaryRecipient();
                } catch (RecipientFormattingException e) {
                    Log.w("MessageNotifier", e);
                    recipient = Recipient.getUnknownRecipient(context);
                }

                Recipients recipients = RecipientFactory.getRecipientsFromMessage(context, message, false);
                long threadId = DatabaseFactory.getThreadDatabase(context).getThreadIdFor(recipients);
                SpannableString body = new SpannableString(context.getString(R.string.MessageNotifier_encrypted_message));
                body.setSpan(new StyleSpan(android.graphics.Typeface.ITALIC), 0, body.length(), Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);

                notificationState.addNotification(new NotificationItem(recipient, recipients, null, threadId, body, null));
            }
        } finally {
            if (reader != null)
                reader.close();
        }
    }

    private static NotificationState constructNotificationState(Context context,
                                                                MasterSecret masterSecret,
                                                                Cursor cursor) {
        NotificationState notificationState = new NotificationState();
        MessageRecord record;
        MmsSmsDatabase.Reader reader;

        if (masterSecret == null)
            reader = DatabaseFactory.getMmsSmsDatabase(context).readerFor(cursor);
        else reader = DatabaseFactory.getMmsSmsDatabase(context).readerFor(cursor, masterSecret);

        while ((record = reader.getNext()) != null) {
            Recipient recipient = record.getIndividualRecipient();
            Recipients recipients = record.getRecipients();
            long threadId = record.getThreadId();
            SpannableString body = record.getDisplayBody();
            Uri image = null;
            Recipients threadRecipients = null;

            if (threadId != -1) {
                threadRecipients = DatabaseFactory.getThreadDatabase(context).getRecipientsForThreadId(threadId);
            }

            // XXXX This is so fucked up.  FIX ME!
            if (body.toString().equals(context.getString(R.string.MessageDisplayHelper_decrypting_please_wait))) {
                body = new SpannableString(context.getString(R.string.MessageNotifier_encrypted_message));
                body.setSpan(new StyleSpan(android.graphics.Typeface.ITALIC), 0, body.length(), Spannable.SPAN_EXCLUSIVE_EXCLUSIVE);
            }

            notificationState.addNotification(new NotificationItem(recipient, recipients, threadRecipients, threadId, body, image));
        }

        reader.close();
        return notificationState;
    }

    private static void setNotificationAlarms(Context context,
                                              NotificationCompat.Builder builder,
                                              boolean signal) {
        String ringtone = TextSecurePreferences.getNotificationRingtone(context);
        boolean vibrate = TextSecurePreferences.isNotificationVibrateEnabled(context);
        String ledColor = TextSecurePreferences.getNotificationLedColor(context);
        String ledBlinkPattern = TextSecurePreferences.getNotificationLedPattern(context);
        String ledBlinkPatternCustom = TextSecurePreferences.getNotificationLedPatternCustom(context);
        String[] blinkPatternArray = parseBlinkPattern(ledBlinkPattern, ledBlinkPatternCustom);

        builder.setSound(TextUtils.isEmpty(ringtone) || !signal ? null : Uri.parse(ringtone));

        if (signal && vibrate) {
            builder.setDefaults(Notification.DEFAULT_VIBRATE);
        }

        if (!ledColor.equals("none")) {
            builder.setLights(Color.parseColor(ledColor),
                    Integer.parseInt(blinkPatternArray[0]),
                    Integer.parseInt(blinkPatternArray[1]));
        }
    }

    private static String[] parseBlinkPattern(String blinkPattern, String blinkPatternCustom) {
        if (blinkPattern.equals("custom"))
            blinkPattern = blinkPatternCustom;

        return blinkPattern.split(",");
    }
}
